multiclass classification in python from scratch